package com.fzq.util;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.pagehelper.PageInfo;
import org.springframework.beans.BeanUtils;

import java.util.List;
import java.util.function.Supplier;
import java.util.stream.Collectors;

/**
 * Bean集合复制工具类
 */
public class BeanListUtils extends BeanUtils {

    /**
     * 转换对象 list
     * @param sources        源对象list
     * @param targetSupplier 目标对象供应方 new
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象list
     */
    public static <S, T> List<T> copyListProperties(List<S> sources, Supplier<T> targetSupplier) {
        if (null == sources || null == targetSupplier) {
            return null;
        }

        return sources.stream().map(s -> {
            T target = targetSupplier.get();
            copyProperties(s, target);
            return target;
        }).collect(Collectors.toList());
    }

    /**
     * mybatis-plus Ipage对象复制
     * @param sources 数据对对应的实体Page对象
     * @param target  复制之后的对象
     * @param <S>     数据库对象实体类
     * @param <T>     目标对象类型(VO)
     * @return 目标对象list
     */
    public static <S, T> IPage<T> copyPage(IPage<S> sources, IPage<T> target, Supplier<T> targetSupplier) {
        // 复制外层
        BeanUtils.copyProperties(sources, target);
        // 复制records对象
        target.setRecords(copyListProperties(sources.getRecords(), targetSupplier));
        return target;
    }

    /**
     * PageHelper对象复制转换PageInfo中的List
     * @param sources 数据对对应的实体Page对象
     * @param target  复制之后的对象
     * @param <S>     数据库对象实体类
     * @param <T>     目标对象类型(VO)
     * @return 转换后的pageInfo
     */
    public static <S,T> PageInfo<T> copyPageInfo(PageInfo<S> sources, PageInfo<T> target, Supplier<T> targetSupplier){
        // 复制外层
        BeanUtils.copyProperties(sources, target);
        target.setList(copyListProperties(sources.getList(), targetSupplier));
        return target;
    }
}
